tensor([[[0.6000, 0.0000],
         [0.8200, 0.0000]],

        [[0.0000, 0.2700],
         [0.1800, 0.5200]],

        [[0.4000, 0.7300],
         [0.0000, 0.4800]]])